/**
 * Copyright (c) 2016-2019 蜂鸟开源 All rights reserved.
 *
 * https://www.hummingbird.io
 *
 * 版权所有，侵权必究！
 */

package io.hummingbird.common.utils;

import com.alibaba.fastjson.JSON;

import org.apache.commons.lang.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.*;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.concurrent.TimeUnit;

/**
 * Redis工具类
 *
 * @author Top
 */
@Component
public class RedisUtils {
	@Autowired
	private StringRedisTemplate stringRedisTemplate;
    @Autowired
    private RedisTemplate redisTemplate;
    @Resource(name="redisTemplate")
    private ValueOperations<String, String> valueOperations;
    @Resource(name="redisTemplate")
    private HashOperations<String, String, Object> hashOperations;
    @Resource(name="redisTemplate")
    private ListOperations<String, Object> listOperations;
    @Resource(name="redisTemplate")
    private SetOperations<String, Object> setOperations;
    @Resource(name="redisTemplate")
    private ZSetOperations<String, Object> zSetOperations;
    /**  默认过期时长，单位：秒 */
    public final static long DEFAULT_EXPIRE = 60 * 60 * 24;
    /**  不设置过期时长 */
    public final static long NOT_EXPIRE = -1;

    public void set(String key, Object value, long expire){
        valueOperations.set(key, toJson(value));
        if(expire != NOT_EXPIRE){
            redisTemplate.expire(key, expire, TimeUnit.SECONDS);
        }
    }

    public Integer increment(String key,long expire){
    	Integer value=get(key, Integer.class);
    	if(null==value){
    		value=1;
    		set(key, value, expire);
    		return value;
    	}
    	Long expireTime=stringRedisTemplate.boundValueOps(key).getExpire();
    	value++;
    	set(key, value, expireTime);
    	return value;
    }
    	public void set(String key, Object value){
    		set(key, value, DEFAULT_EXPIRE);
    	}

    	public <T> T get(String key, Class<T> clazz, long expire) {
    		String value = valueOperations.get(key);
    		if(expire != NOT_EXPIRE){
    			redisTemplate.expire(key, expire, TimeUnit.SECONDS);
    		}
    		return value == null ? null : fromJson(value, clazz);
    	}

    	public <T> T get(String key, Class<T> clazz) {
    		return get(key, clazz, NOT_EXPIRE);
    	}

    	public String get(String key, long expire) {
    		String value = valueOperations.get(key);
    		if(expire != NOT_EXPIRE){
    			redisTemplate.expire(key, expire, TimeUnit.SECONDS);
    		}
    		return value;
    	}

    	public String get(String key) {
    		return get(key, NOT_EXPIRE);
    	}

    	public void delete(String key) {
    		redisTemplate.delete(key);
    	}

    	/**
    	 * Object转成JSON数据
    	 */
    	private String toJson(Object object){
    		if(object instanceof Integer || object instanceof Long || object instanceof Float ||
    				object instanceof Double || object instanceof Boolean || object instanceof String){
    			return String.valueOf(object);
    		}
    		return JSON.toJSONString(object);
    	}

    	/**
    	 * JSON数据，转成Object
    	 */
    	private <T> T fromJson(String json, Class<T> clazz){
    		return JSON.parseObject(json, clazz);
    	}
}
